package com.github.kaelthasbmg.bean;

import java.util.ArrayList;
import java.util.List;

/**
 * 主键定义
 *
 * @author Carsymor
 */
public class PrimaryKey {

    /**
     * 主键所属表名
     */
    private String tableName;

    /**
     * 主键包含列
     */
    private List<String> columns;

    public String getTableName() {
        return tableName;
    }

    public void setTableName(String tableName) {
        this.tableName = tableName;
    }

    public List<String> getColumns() {
        return columns;
    }

    public void setColumns(List<String> columns) {
        this.columns = columns;
    }

    /**
     * 获取主键名称
     * @return 主键名称
     */
    public String getKeyName() {
        return "PK_" + this.tableName;
    }

    /**
     * 添加主键列
     * @param columnName 列名
     */
    public void addColumn(String columnName) {
        if (this.columns == null) {
            this.columns = new ArrayList<>();
        }
        this.columns.add(columnName);
    }
}
